Stock Transaction program in python